#统计目录文件大小

import os
summary=0
def size_file(str2):
    #获取文件的大小并进行累加
    global summary
    summary=summary+os.path.getsize(str2)
def size_dir(str1):
    #获取目录所有文件及目录大小信息，并进行迭代相加
    dlist=os.listdir(str1)
    print("正在统计目录："+str1)
    for f in dlist:
        #获取每个文件路径
        file1=os.path.join(str1,f)
        if os.path.isfile(file1):
            size_file(file1)
        else:
            size_dir(file1)
def init_os(path_init):
    #初始化,判断路径是否存在,若存在，则进行统计
    if os.path.exists(path_init):
        size_dir(path_init)
        if summary >1024:
            print("文件大小为:"+str(summary/1024)+"KB")
        elif summary>1000000:
            print("文件大小为："+str((summary/1024)/1024)+"MB")
        else:
            print("文件大小为:"+str(summary)+"B")
    else:
        print("目录不存在!,请重新输入.")
path=input("输入目录：")
init_os(path)

